解决numpy.core._exceptions.MemoryError: Unable to allocate 1.04 MiB for an array